project(1.3-cnn-and-dynamic-shape)

set(BASE_PATH /media/chyang/data/third_party/linux)
set(CUDA_HOME ${BASE_PATH}/cuda/cuda_12.3)
set(TRT_HOME ${BASE_PATH}/tensorrt/TensorRT-8.6.1.6/)

# Find CUDA.
include_directories(${CUDA_HOME}/include)
link_directories(${CUDA_HOME}/lib)

# Find TensorRT
include_directories(${TRT_HOME}/include/)
link_directories(${TRT_HOME}/lib)

include_directories(${PROJECT_SOURCE_DIR})
include_directories(/usr/include/opencv4)

include_directories(${PROJECT_SOURCE_DIR}/src)

add_executable(main src/main.cpp)
target_link_libraries(main PUBLIC cuda cudart nvinfer nvinfer_plugin nvonnxparser libopencv_core.so libopencv_highgui.so libopencv_imgproc.so libopencv_video.so libopencv_videoio.so libopencv_imgcodecs.so)